package com.zjj.algorithm.learning.leetcode.tree;

/**
 * 110. 平衡二叉树 简单题
 * 给定一个二叉树，判断它是否是高度平衡的二叉树。
 * <p>
 * 本题中，一棵高度平衡二叉树定义为：
 * <p>
 * 一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
 *
 * @author zjj_admin
 * @date 2022/12/20 10:19
 */
public class BalancedBinaryTree {

    public static void main(String[] args) {

        TreeNode node1 = new TreeNode(1);
        TreeNode node2 = new TreeNode(2);
        TreeNode node3 = new TreeNode(3);
        TreeNode node4 = new TreeNode(4);

        node1.left = node2;
        node2.left = node3;
        node2.right = node4;

        boolean res = isBalanced(node1);
        System.out.println("res = " + res);

    }


    /**
     * 时间
     * 0 ms
     * 击败
     * 100%
     * 内存
     * 41.2 MB
     * 击败
     * 47.33%
     *
     * @param root
     * @return
     */
    public static boolean isBalanced(TreeNode root) {
        return height(root) >= 0;
    }

    private static int height(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int l = height(root.left);
        int r = height(root.right);
        if (l >= 0 && r >= 0 && Math.abs(l - r) <= 1) {
            return Math.max(l, r) + 1;
        } else {
            return -1;
        }
    }
}
